Overall Strategy

alt text

The main idea was to make a model as simple as possible without the complication of preprocessing or postprocessing. For training the model, I used only one image of a ball which I found using a simple OpenCV function. After this, I pasted the same ball onto background photos I took from Kaggle. I divided the ball into 2 classes: magic-ball and a fixed color circle/ball. There are other wide-ranging possibilities to perform the task, but I chose this way because, in my understanding, it is the fastest (as you can see, it takes me an average of 70 ms per image) and requires minimum time for pre- and post-processing.

Datastes and requirements

step 1

step 2

Train Yolov8 nano

Yolo results

Evaluation

Conclusions

alt text

alt text